import os, sys
sys.path.append(os.getcwd())
from configs._base_ import DATASET_CFG_VOCAUG_512x512, DATALOADER_CFG_BS16
from configs.fcn.base_cfg import SEGMENTOR_CFG
import copy

SEGMENTOR_CFG = copy.deepcopy(SEGMENTOR_CFG)  # deepcopy
SEGMENTOR_CFG['dataset'] = DATASET_CFG_VOCAUG_512x512.copy() # modify dataset config
SEGMENTOR_CFG['dataloader'] = DATALOADER_CFG_BS16.copy() # modify dataloader config
SEGMENTOR_CFG['scheduler']['max_epochs'] = 60 # modify scheduler config
# modify other segmentor configs
SEGMENTOR_CFG['num_classes'] = 21
SEGMENTOR_CFG['backbone'] = {
    'type': 'ResNet', 'depth': 50, 'structure_type': 'resnet50conv3x3stem',
    'pretrained': True, 'outstride': 16, 'use_conv3x3_stem': True, 'selected_indices': (2, 3),
}
SEGMENTOR_CFG['eval_interval_epochs'] = 1
SEGMENTOR_CFG['save_interval_epochs'] = 5
SEGMENTOR_CFG['work_dir'] = 'fcn_resnet50os16_voc_test'
SEGMENTOR_CFG['logfilepath'] = 'fcn_resnet50os16_voc_test/fcn_resnet50os16_voc.log'
SEGMENTOR_CFG['resultsavepath'] = 'fcn_resnet50os16_voc_test/fcn_resnet50os16_voc_results.pkl'
